#include <iostream>
#include <sys/types.h>
#include <sys/socket.h>
#include <cerrno>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <string.h>

//const uint16_t port = 8080;

void Usage(std::string proc)
{
    std::cout << "Usage \n\t" << proc << "  server_port" << std::endl;
}

int main(int argc, char* argv[])
{
    if(argc != 2)
    {
        Usage(argv[0]);
        return 0;
    }
    uint16_t port = atoi(argv[1]);
    //创建套接字，打开网络文件
    int sock = socket(AF_INET, SOCK_DGRAM, 0);
    if(sock < 0)
    {
        std::cerr << "socket create error" << errno << std::endl; 
        return 1;
    }

    //2.服务器绑定端口和IP
    //服务器的IP加port必须被客户知道
    struct sockaddr_in local;
    local.sin_family = AF_INET;
    local.sin_port = htons(port);//此处的端口号，是我们计算机上的变量，是主机序列
    // a.需要将人识别的点分十进制，字符串风格的IP地址转换成为4字节的整形IP
    // b.也需要考虑大小端
    // in_addr_t inet_addr (const char *__cp) 能完成上面两个工作
    // 云服务器，不允许用户直接绑定公网IP，另外，我们正常编写的时候也不会指明IP
    //local.sin_addr.s_addr = inet_addr("43.431.43.43")// 
    //INADDR_ANY 如果你绑定的是确定的IP（主机），意味着只有发到该IP主机上的数据才会交给你的网络进程
    //但是，一般的服务器可能有多张网卡，配置多个IP，我们需要的不是某个IP上的数据，而是所有发送到该主机上的数据
    local.sin_addr.s_addr = INADDR_ANY; //#define INADDR_ANY ((in_addr_t) 0x00000000)

    if(bind(sock, (struct sockaddr*)&local, sizeof(local)) < 0)
    {
        std::cerr << "bind error"  << errno << std::endl;
        return 2; 
    }

    // 提供服务
    bool quit = false;
    #define NUM 1024
    char buffer[NUM];
    while(!quit)
    {
        struct sockaddr_in peer;
        socklen_t len = sizeof(peer);
        ssize_t cnt = recvfrom(sock, buffer, sizeof(buffer) - 1, 0, (struct sockaddr*)&peer, &len);
        if(cnt > 0)
        {
            buffer[cnt] = 0;
            FILE* fp = popen(buffer, "r");
            if(fp == nullptr)
            {
                std::cout << "open failed" << std::endl;
                return 3;
            }
            std::string s;
            char msg[1024] = { 0 };
            while(fgets(msg, sizeof(msg), fp) != nullptr);
            {
                s += msg;
            }
            sendto(sock, s.c_str(), s.size(), 0, (struct sockaddr*)&peer, len);

            // char msg[1024] = { 0 };
            // size_t number = fread(msg, sizeof(char), strlen(msg), fp);
            // if(number > 0)
            // {
            //     msg[number] = 0;
            //     std::cout << "client: " << buffer << std::endl;
            //     //std::string s("server say hello");
            //     //sendto(sock, s.c_str(), s.size(), 0, (struct sockaddr*)&peer, len);
            //     sendto(sock, msg, strlen(msg), 0, (struct sockaddr*)&peer, len);
            // }
            // else
            // {
            //     // TODO 
            // }
            pclose(fp);
        }
        else
        {
            // TODO
        }
    }
    return 0;
}
